Adding main method check and argparse.