water bucket using python