Seems to be working great.
Trying out more efficient representation of one-hot units