Whoops forgot something, sorry. I also added greying-out of the selection button