Adding updated main.go so build can pass.