Add pyyaml as dependency