Improving our code through testing