Code for neural multiplexing project