add tests for what I changed