Update TODO with additional methods