Test existing code and add to methods so tests pass.