Fixed problem with css classes