using flex box instead of float