Bug Fixes

Few Bugs were fixed!!
Need to make the neural network efficient.