Updated studio.py with new code for pygame