Make context size configurable