Modify input screens for menu selection