Quick tidy up of docstrings and reset beginner stage to match handouts.