Lab 10: 2 of 3 (bug fixed)