Small math/algorithm project in Go