#17 add test