correction to small bug
removal of unnecessary code