Adding __main__ module