Revert "Remove yaml dependency - it's not needed"

This reverts commit 0ba32f303204f00e49b4f427e85fbe148b847851.