updated nd4j api