update main.go
add template code