modify check removeable balls logic