Simplifying code for testing purposes