Move balloon creation to function