Small change to getting selection rects