Assignment 2

Made a change in the intending of code