Made sure docstrings matched code

Changed j to index everywhere