Added more flex box classes