Updated sample program in main.go