Support yaml extension
remove extra spaces (style)
Add test for yaml extension