#16 add test