Add methods in states controller.