Remove unused import
Updated documentation