switching broadcasting to use axis=None