Move __main__ script into another file