Update main.go

Added code for chapter 4.3