Including test code to the build