work on controller & model