simple code using arrays in go