Add TypeContext class