Update MS demo with api changes