training screen bugs fixed