diff --git a/dalle2_pytorch/cli.py b/dalle2_pytorch/cli.py
index 5b60af9..ed87e8b 100644
--- a/dalle2_pytorch/cli.py
+++ b/dalle2_pytorch/cli.py
@@ -2,3 +2,8 @@ import click
 
 def main():
     pass
+
+@click.command()
+@click.argument('text')
+def dream(text):
+    return image
diff --git a/setup.py b/setup.py
index 05216f8..1672c3a 100644
--- a/setup.py
+++ b/setup.py
@@ -7,6 +7,7 @@ setup(
   entry_points={
     'console_scripts': [
       'dalle2_pytorch = dalle2_pytorch.cli:main',
+      'dream = dalle2_pytorch.cli:dream'
     ],
   },
   version = '0.0.2',