Added junit test