Issue : Typo in `01_weight_zero_learning_rate.py`

[PrinceSajjadHussain]

There's a typo in the line ax[0].scatter(X_train[y_trai == 1, 0], X_train[y_train == 1, 1],. y_trai should be y_train. This will cause an error.