fixed bug with coloring
final commit ;)