update to prog assign2

cleaned up the formatting and added a few more comments