updated to use correct mass! was passing stellar mass but still multiplying by the baryon fraction