Fixed import error - updated documentation