Fix CSS issue, add missing class