Update flex styles