Model building steps added