Updated styles

Switched to flexbox