add documentation and update some imports