diff --git a/screenshot.png b/.github/screenshot.png
similarity index 100%
rename from screenshot.png
rename to .github/screenshot.png
diff --git a/.gitignore b/.gitignore
index b76be59..c619c0b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -23,7 +23,7 @@ lerna-debug.log*
 .vscode
 
 # Distribution
-bin
+dist
 
 # Eslint cache
 .eslintcache
diff --git a/README.md b/README.md
index 0fa8d38..9feb3c3 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 <div align="center">
   <div>
-    <img src="https://raw.githubusercontent.com/Nutlope/aicommits/main/screenshot.png" alt="AI Commits"/>
+    <img src=".github/screenshot.png" alt="AI Commits"/>
     <h1 align="center">AI Commits</h1>
   </div>
 	<p>A CLI that writes your git commit messages for you with AI. Never write a commit message again.</p>
diff --git a/package.json b/package.json
index 05124d6..3a8c94e 100644
--- a/package.json
+++ b/package.json
@@ -3,9 +3,9 @@
   "version": "1.0.5",
   "description": "Writes your git commit messages for you with AI",
   "files": [
-    "bin"
+    "dist"
   ],
-  "bin": "bin/index.js",
+  "bin": "dist/index.js",
   "repository": "https://github.com/Nutlope/aicommits",
   "author": "Hassan El Mghari (@nutlope)",
   "license": "MIT",
diff --git a/aicommits.ts b/src/aicommits.ts
similarity index 100%
rename from aicommits.ts
rename to src/aicommits.ts
diff --git a/index.ts b/src/index.ts
similarity index 100%
rename from index.ts
rename to src/index.ts
diff --git a/tsconfig.json b/tsconfig.json
index 510440a..5061864 100644
--- a/tsconfig.json
+++ b/tsconfig.json
@@ -2,9 +2,10 @@
   "compilerOptions": {
     "target": "es2016",
     "module": "commonjs",
-    "outDir": "./bin",
+    "outDir": "./dist",
     "esModuleInterop": true,
     "forceConsistentCasingInFileNames": true,
     "skipLibCheck": true /* Skip type checking all .d.ts files. */
   },
+  "include": ["src"]
 }