Add documentation
Remove unnecessary import