Minor fixes + changed the default number of epochs to 100.