clean up __main__