Add minimum code required to pass test