Adding code which was used for the first demo