Updated yaml and docs

Updated yaml and docs