Adding tests units