formatting and correcting docstrings
docstring and remove one_hot from mnist