Can you add a memory management test for this class too?