Corrected import and documentation