Slight optimization

No need to perform computations for `DIST_COLOR` if we're not using it.
This might make not a big difference in speed, but maybe it does.