changes done for second part of assignment