Updated main.go and main_test.go.