Added Unit Training

There are still a few tweaks and minor implementations to make.