Add collapse methods to selection handler