fix migration script, add some colors