Add more tests for shell coloring