Add PyGame support