I tried to follow the testing style but I'm not quite sure if this is how you'd do it. Happy to change this around if there's a way you'd prefer.