Refactored for loop and ran 'fmt' on both files

go fmt : http://blog.golang.org/go-fmt-your-code !