Added controller and mark function