Fixes yaml dependency

Using PyYAML instead of yaml