Update library code to handle multi-color displays