#2 Adding test with highlighting