Added the class in the builder