move context to separate module