Training updates

Needs error handling and feedback messages