Updated code based on go fmt