added flex styling