Have training pretty much working, need to fix up normalization issues