Add methods in main controller