Adds functionality to relay_test