fix homework 5