First test didnt work. I was calling to return the wrong variable. Fixed to the right variable and ready to test.