Adding programming assignment 2