update main.go so can run with new code(oneway)