update main.go to work with new server package