Update jumble.py

clean up the code