add a __main__ check so this file can be used by a UI driver