Update main.go

updated the import to be local