Add neural network clustering work