diff --git a/Makefile b/Makefile
new file mode 100644
index 0000000..8e18ca8
--- /dev/null
+++ b/Makefile
@@ -0,0 +1,2 @@
+fmt:
+	find . -name '*.go' -exec gofumpt -w -s -extra {} \;
\ No newline at end of file
diff --git a/cmd/github-site-gen/main.go b/cmd/github-site-gen/main.go
index 048a041..95ae08c 100644
--- a/cmd/github-site-gen/main.go
+++ b/cmd/github-site-gen/main.go
@@ -93,7 +93,7 @@ func main() {
 	repoURL := fmt.Sprintf("https://%s/%s/%s.git", *githost, owner, repo)
 
 	// Create output directory if it doesn't exist
-	if err := os.MkdirAll(*outputFlag, 0755); err != nil {
+	if err := os.MkdirAll(*outputFlag, 0o755); err != nil {
 		log.Fatalf("Failed to create output directory: %v", err)
 	}
 
@@ -109,7 +109,7 @@ func main() {
 		defer os.RemoveAll(tempDir) // Clean up when done
 	} else {
 		// Ensure the specified work directory exists
-		if err := os.MkdirAll(workDir, 0755); err != nil {
+		if err := os.MkdirAll(workDir, 0o755); err != nil {
 			log.Fatalf("Failed to create working directory: %v", err)
 		}
 	}
diff --git a/pkg/generator/generator.go b/pkg/generator/generator.go
index 20923af..5bbd0ca 100644
--- a/pkg/generator/generator.go
+++ b/pkg/generator/generator.go
@@ -74,7 +74,7 @@ func (g *Generator) GenerateSite() (*GenerationResult, error) {
 
 	// Create docs directory
 	docsDir := filepath.Join(g.outputDir, "docs")
-	if err := os.MkdirAll(docsDir, 0755); err != nil {
+	if err := os.MkdirAll(docsDir, 0o755); err != nil {
 		return nil, fmt.Errorf("failed to create docs directory: %w", err)
 	}
 
@@ -85,7 +85,7 @@ func (g *Generator) GenerateSite() (*GenerationResult, error) {
 
 	// Create image directory if needed
 	imagesDir := filepath.Join(g.outputDir, "images")
-	if err := os.MkdirAll(imagesDir, 0755); err != nil {
+	if err := os.MkdirAll(imagesDir, 0o755); err != nil {
 		return nil, fmt.Errorf("failed to create images directory: %w", err)
 	}
 
@@ -226,7 +226,7 @@ func (g *Generator) generateMainPage(docsPages []utils.DocPage) error {
 
 	// Write to file
 	outputPath := filepath.Join(g.outputDir, "index.html")
-	if err := os.WriteFile(outputPath, buf.Bytes(), 0644); err != nil {
+	if err := os.WriteFile(outputPath, buf.Bytes(), 0o644); err != nil {
 		return fmt.Errorf("failed to write index.html: %w", err)
 	}
 
@@ -288,12 +288,12 @@ func (g *Generator) generateDocPage(path, content string, docsPages []utils.DocP
 
 	// Ensure output directory exists
 	outPath := filepath.Join(g.outputDir, outputPath)
-	if err := os.MkdirAll(filepath.Dir(outPath), 0755); err != nil {
+	if err := os.MkdirAll(filepath.Dir(outPath), 0o755); err != nil {
 		return fmt.Errorf("failed to create directory for %s: %w", outPath, err)
 	}
 
 	// Write to file
-	if err := os.WriteFile(outPath, buf.Bytes(), 0644); err != nil {
+	if err := os.WriteFile(outPath, buf.Bytes(), 0o644); err != nil {
 		return fmt.Errorf("failed to write %s: %w", outPath, err)
 	}
 
@@ -387,7 +387,7 @@ func copyFile(src, dst string) error {
 func GenerateRootStyle(outputDir string) error {
 	// write the templates.StyleTemplate to the root of the output directory
 	stylePath := filepath.Join(outputDir, "style.css")
-	if err := os.WriteFile(stylePath, []byte(templates.StyleTemplate), 0644); err != nil {
+	if err := os.WriteFile(stylePath, []byte(templates.StyleTemplate), 0o644); err != nil {
 		return fmt.Errorf("failed to write style.css: %w", err)
 	}
 	return nil