Addes unit tests