Adding application-specific checkpointing for the demo app