added two models but don't finished yet