add tests for edited code