Programming Assignment-2 minor fix