Added more junit