Add few functions to the controller