Add ability to set selection