Added argument parsing with argparse
Added comments