Removed some lines, cleaned up docstring.