Add more tests, change tests to DRY