Use FirstN and partititonAggregate correctly... maybe?