1. Changed screen layout to train and recognize
2. Refactored code for alot of reuse
3. Add more screens and stability