python basal ganglia model