Update solution.go