Updates Demo project
Updates main class to fix #26
Update main class to fix #26