Ensure consistent docstrings

[Sparks29032]

Problem

Make sure docstrings for the functions are consistent across the repository. We want to test how it looks in two places.

1. Check on PyCharm / any IDE that the docstrings that display when hovering over it makes sense.
2. Check after the docstrings are built that the documentation looks how we want it.

Either we match everything to morphsqueeze or the other way, whichever looks better.

Proposed solution

[Sparks29032]

@Luiskitsu