changing if __name__ == __main__: snippet to avoid lint errors