Fixed imports

Imports no longer broken after moving function from plot.go into main.go.