Update OkPi.py

made it easier to change things