Updated layout to flat