Cleaned up by go fmt