Added missing methods in protocol