diff --git a/Tensorflow-2.0-Example.py b/Tensorflow-2.0-Example.py
index 8c980c9..a6da60e 100644
--- a/Tensorflow-2.0-Example.py
+++ b/Tensorflow-2.0-Example.py
@@ -3,6 +3,8 @@
 import numpy as np
 import random
 from sklearn.model_selection import train_test_split
+from tensorflow import feature_column
+from tensorflow.keras import layers
 
 
 Num = 5000