changed __main__.py