Add index_to_one_hot module and switch design to use it