Using flexbox for better layout