Fixed code to return the number input if the input is only 1 number