Python method chaining in functional programming style