Update program for #2