Added robot coloring. Fixes #17