removed unused import
Updated documentation