Changing test output + return data