That's better. The code I wrote was redoing a 'correct' Java implementation again in Go. So please make sure the change you made was doing the same thing. You can change it yourself.