Update context_lib.py