updates for programming assignment 2