Got first test working