modified Model.h