Add documentation, cleaup unused code