Add missing methods from demo