Add better tests.