Added followup demo code