Updated menu and file choosing code