Changed main.go to collect