diff --git a/Makefile b/Makefile
index bcc1af9..120543d 100644
--- a/Makefile
+++ b/Makefile
@@ -3,5 +3,12 @@ deploy:
 	cd ui; make deploy
 	cd infra; make deploy
 
+image: SHELL := /bin/bash
 image:
-	codevis -i ./ --whitelist-extension go,hcl,tf,css,js,templ,
+	[[ -d "/tmp/capstone" ]] || mkdir /tmp/capstone
+	cp -R infra/ /tmp/capstone/
+	cp -R service/ /tmp/capstone/
+	cp -R ui/ /tmp/capstone/
+	rm -rf /tmp/capstone/ui/templates/*_templ.go
+	codevis -i /tmp/capstone --whitelist-extension go,hcl,tf,templ -o ./out.png
+	rm -rf /tmp/capstone/*
diff --git a/code.png b/code.png
index c53b371..933328e 100644
Binary files a/code.png and b/code.png differ