Work in progress for neural net shenanigans