Adding more functionality to add/remove selected items