Added test for code highlighting