Updated main.go in order to use the correct import path