Improve random color function