Cleaning up demo js