minor correction in margin growing code