Change layout to flexbox instead of floats