Add support for yml extension

Also added the test case.