added flex box for layout