better way to check user's choice