Correct lstm docstring and return.