Add missing @return tag, toString() update