+project2.go
update for problem 3