added some Junit testing