My bad
Added while statement