#4 add test