Fixed minor errors + added the space for docstrings to be done