changes done w.r.t. review