update to prog assign2

a few minor changes