we should pass the context to this class and reuse it of not None, instead of creating a second one