Update main.go

modified import to match git repo