Writing documentation, tidying up classes.