Correct input funciont