Fix dependency error

Moved to using pyyaml, but never added it to dependencies.