Change the code for programming assignment 2 by Archit

Both the parts were completed and submitted