Updates to simple demo