Update color generation

Last commit was incomplete. oops