find_balloon: remove duplicate code from main