Add Unit test for selection