Add Context class