Modified code style according to PEP8, added argument parser