Add support for scaffold generator