some changes to second assignment