Div classes fix