removed unused imports in colab notebook