Completed the neural nets for the classes