simplify main.go