Add training button functionality
Merge branch 'training'