updates to programming assignment 1