last little fix for homework 3