Update main.go, smårydding