Python reservoir script