Creating test problem to test one piece of code at a time