updates for assignment-1