Change width function from flex-grid to percentage