Improve color

solve #7

note: please merge after #98 because it's relying on it.