Add junit test