Optimization of the base class model. (currently)