updates for toy example