add argparse
add print statement