imagine.py minor changes in __main__()