Implement missing methods in controller