Add "remove" param for setStyle method to be able to remove classes