Tidying up in code documentation