Enclose in __main__ section