Fixed learn_yz_x_ss.py