Minor refactor to Go version to make it shorter.

Use `time.Since`, which is shorthand for `time.Now().Sub`.