Add styling for new flexible layout