Some refactor
.addClass(function)