add rivet code