Add Context yaml file
Change test to use context.yml