Update Chocolate Feast

Improved answer with while loop