Water Jug In Python