Simple Tensorflow Code In Python