diff --git a/blocks/image/server.ts b/blocks/image/server.ts
index 6c6adce..81c3f7f 100644
--- a/blocks/image/server.ts
+++ b/blocks/image/server.ts
@@ -1,11 +1,10 @@
 import { myProvider } from '@/lib/ai/models';
 import { createDocumentHandler } from '@/lib/blocks/server';
-import { saveDocument } from '@/lib/db/queries';
 import { experimental_generateImage } from 'ai';
 
 export const imageDocumentHandler = createDocumentHandler<'image'>({
   kind: 'image',
-  onCreateDocument: async ({ id, title, dataStream, session }) => {
+  onCreateDocument: async ({ title, dataStream }) => {
     let draftContent = '';
 
     const { image } = await experimental_generateImage({
@@ -21,23 +20,13 @@ export const imageDocumentHandler = createDocumentHandler<'image'>({
       content: image.base64,
     });
 
-    if (session?.user?.id) {
-      await saveDocument({
-        id,
-        title,
-        kind: 'image',
-        content: draftContent,
-        userId: session.user.id,
-      });
-    }
-
     return draftContent;
   },
   onUpdateDocument: async ({ description, dataStream }) => {
     let draftContent = '';
 
     const { image } = await experimental_generateImage({
-      model: myProvider.imageModel('image-model'),
+      model: myProvider.imageModel('small-model'),
       prompt: description,
       n: 1,
     });
diff --git a/components/chat.tsx b/components/chat.tsx
index 3922b8f..6911e66 100644
--- a/components/chat.tsx
+++ b/components/chat.tsx
@@ -52,7 +52,6 @@ export function Chat({
       mutate('/api/history');
     },
     onError: (error) => {
-      console.log(error);
       toast.error('An error occured, please try again!');
     },
   });
diff --git a/components/messages.tsx b/components/messages.tsx
index 4ba90b2..1c14b2f 100644
--- a/components/messages.tsx
+++ b/components/messages.tsx
@@ -38,7 +38,7 @@ function PureMessages({
       ref={messagesContainerRef}
       className="flex flex-col min-w-0 gap-6 flex-1 overflow-y-scroll pt-4"
     >
-      {/* {messages.length === 0 && <Overview />} */}
+      {messages.length === 0 && <Overview />}
 
       {messages.map((message, index) => (
         <PreviewMessage
diff --git a/lib/ai/models.ts b/lib/ai/models.ts
index cbfc0a1..adb8dcd 100644
--- a/lib/ai/models.ts
+++ b/lib/ai/models.ts
@@ -20,7 +20,8 @@ export const myProvider = customProvider({
     'block-model': openai('gpt-4o-mini'),
   },
   imageModels: {
-    'small-model': openai.image('dall-e-3'),
+    'small-model': openai.image('dall-e-2'),
+    'large-model': openai.image('dall-e-3'),
   },
 });