Refactored specs to use context more.