Add use statement of Yaml Formatter class

This PR fixes #124 issue