Adding custom code execution to the test project