Changed demo code format