* Added some javadoc
* Printing out the training error as well