Add PyYAML dependency