Fixed critical bug
removed unused import
minor documentation update