altered logic for test menu