Clean up init.go