Add demo code from class handout