Fixed spacing between classes and methods and updated docstring