Cleanup input validation
Use argparse to parse the integer as an integer