Understanding neural network units in AI