working through softmax_exercise