Add color script
Refactor color a bit