Allan please add code to make tests pass