Fixed Hw 5.4 answer