Update interesting_puzzles/dropbox_diet.py

Better code and documentation