Add check to see if input is correct