Add controller related functions