Fix non_zero_rand() and remove dot notation in imports because apparently they don't work?