Can you add accompanying tests for all the changes?