Updated context module