Added contextual classes to query run states