fix for last homework