2 files changed, 3 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py
index ca499a3..9757367 100644
--- a/cli/app/commands/biggan/search_class.py
+++ b/cli/app/commands/biggan/search_class.py
@@ -1,5 +1,7 @@
 import click
 
+import os
+import time
 from app.search.search_class import find_nearest_vector_for_images
 
 @click.command('')
diff --git a/cli/cli.py b/cli/cli.py
index 4e3fdb2..bcfda6b 100755
--- a/cli/cli.py
+++ b/cli/cli.py
@@ -52,7 +52,7 @@ if __name__ == '__main__':
       help='Verbosity: -v DEBUG, -vv INFO, -vvv WARN, -vvvv ERROR, -vvvvv CRITICAL')
     @click.pass_context
     def cli(ctx, opt_verbosity):
-      """\033[1m\033[94mMEGAPIXELS\033[0m                                                
+      """\033[1m\033[94mBigGAN Search\033[0m                                                
       """
       ctx.opts = {}
       logger_utils.Logger.create(verbosity=opt_verbosity) # init logger