Add if else to prevent cheating