Update for programming assignment 3