Fixed putWater method

putWater method works correctly now. Added 2 todo in the code