Programming Assignment-2 Updates

Updated required code as part of Programming Assignment-2.