rewrite main.go