Added code and test to Go