Updated main menu. Updated class diagram.