improved on the set up and main menu