Update setup.py

added PyYAML as dependency