Added method in controller