Update TODO - tests good enough for now