Trying to get the first test to pass, wip