Add missing @return annotations