Adding trainer and brain

Still work in progress