add __name__ check