Update setup.py to include install requirement of pygame