Updated get_context to always include max rated nodes