Added if statement to check for int