Done with programming assignment 2