Another improvement for mainMenu