Solution for 021 in Go