Adding testhat tests