Added flex to ensure columns expand with container

#114