added another state. For more effectful removal of the balls