Add sorted set tests