Added find_ball code from Alex.