update with junit test