small change
small bug corrected