Simplified juxt doctest

Also removed the groupby anti-pattern.