Move context into separate file