add puts to unit test