Adding the code changes for the programming assignment