Set controller in init method (instead of ready)