added flex box instead of float