Add support for yaml configuration