Tensorflow Simple Code