Wrote tests for context