demo updates + bug fixes