Fixed some docstring issues