Added tests and method for partitioning a collections around an age