better __main__ setup in pypy