Update cleanup.go