Minor refactor to Go version to make it shorter.

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