minor fixes in main.go