Fixing `ContextError`
Adding basic test framework