Cleaned up script argument structure with argparse