Generalize the type of step in ParLoop from Int to Expression.