Got test working