change link class
clean up code