minor changes and added yaml