Added model to controller