Remove unused code, add docs