Modify demo page to include initial condition functionality