updated assign-2