Updated documentation, removed unnecessary imports.