modify the register to simple model