Put yaml back into setup.py