Added link click event to change class