Fix deprecation warning
Add documentation comments