Adding tests for 'with'