Week3-Assignment-2

Some improvements