Clean up Rabi example