Problem 2 is now done in Go