Fixed the bugs. The training session screen is borked now though.