Added functionality to change/remove styles