Model changes: more hidden layers, miscellaneous training updates