added flex to the columns