Updated the addSelectionClass to be more dynamic by taking a selectionClass param