Changed the input to take ints