TreeSet now works as wanted :)