Use argparse for better cli usage