adds some changes to main.go