Update SoftMax.py