Adding source code for completed demo