Add demonstration code