added setter for context to draw to in UpdateModule