Edit: Move argparser into the __name__ == '__main__' part