<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def prox_unity_plus(X, step, axis=0):
 """Non-negative projection onto sum=1 along an axis """

    return prox_unity(prox_plus(X, step), step, axis=axis)