Started adding assertions to tests