small layout change to be consistent with demo just recorded