#8 add test