Multi-Class Classification using Tensorflow