Update main.go

Added header.