Add method to return a set