added coloring, it works :)