Added junit and wrote first unittest