Yes I expected there to be a button to put just the balls back. Easy enough with these to remember which multipliers you want to keep where but with more complex puzzles that'd be trickier.
