Further implementation and testing for YAMLDataSource