Update css - fix wrap styles