v0.3

Added train functionality and a bit of comment.
Only need testing with a simulator and cleaning up