Add check for correct input.