Update assign2a.s