homework 5 program4 second fix