adding expect tests per code review