Including tests in updated code