Add controller implementation