More docstrings
Stage 1 seems to work, unsure whether it's saving anything though