Improving documentation.

Little change to fix the import line on documentation. :)