Add some documentation and remove unused modules