Add notes re: things to add in assign_brain_states.