added to controller

added to controller