class move to controller