add update to homework week4 (in work)