Add support for yml extension