Add habitat for editor and reshuffle when options are checked to before initializing the display window.

import yourgameshortname.main
if __name__ == "__main__":
    yourgameshortname.main.main()