Refactor using yang extention codegen