Command line argument parsing using argparse
Removed unused import