updated package to use yaml instead of annotation