Add docstrings

Created by: asogaard

We should aim to document all non-trivial methods/functions and modules, since these are used to automatically generate the API documentation at https://icecube.github.io/graphnet. While this is not something that need to be done in one go, we should aim to have all new PRs include sufficient documentation, and to add documentation when reworking parts of the code where it's missing.