Add these tests: