minor improvements
Merge branch 'nikhil' of https://github.com/nikhiljjoshi/NeuralNetwork into nikhil