Fix up vanilla neural net by normalizing data and stuff