Update assign2.s