Add Junit Tests