Fixed typo in comment
Wrote unit test for LogitModel.java
Minor edits