add code to make tests pass