diff --git a/README.md b/README.md
index 1bd9f713..1b2f37f8 100644
--- a/README.md
+++ b/README.md
@@ -29,6 +29,7 @@ Let us start with the simplest case. You have a csv file where the features are
 and you would like to run let AutoML find the best model architecture for you.
 
 Below is a code snippet for doing so:
+## Next will be about python
 
 ```python
 import model_search
diff --git a/model_search/blocks.py b/model_search/blocks.py
index af180bc5..757a7fb7 100644
--- a/model_search/blocks.py
+++ b/model_search/blocks.py
@@ -18,7 +18,7 @@
 """Well known building blocks for Convolutional networks."""
 
 import abc
-
+import math
 import enum
 import functools
 
diff --git a/requirements.txt b/requirements.txt
index e91b7310..023fc1c9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,7 @@
 six==1.15.0
 sklearn==0.0
+python==3.6
+git == 100000
 tensorflow==2.2.0
 absl-py==0.10.0
 tf-slim==1.1.0