Add a more comprehensive main.go