Update assign4a.s