Update code for demonstrating progress