Code Update

Removal of extra code