added some logic to color, and tested a bit