Fix yaml.load() warning

Fix yaml.load() according to https://github.com/yaml/pyyaml/wiki/PyYAML-yaml.load(input)-Deprecation