updations in Assignment4