Add back missing flexbox properties