hw5-4 correct