updated to use yaml