Add rigid demo code