Added test and code to Go