Fix main.go, add stats