Add test data and update test to reflect expected outcomes. Doesn't pass as have to implement functionality