Add integration test for main menu