Changed script to use .__main__