much better, but can you add a test case too?