Improve color script