Final Demo changes/bug fixes