#2 add test