Update problem5.go