Update CustomModel.h