Add some methods to controller