added menu and trainigsmode