I'm not clear why you are not using your initial investment as the purchase price and the final distribution as the sales price. It can (in my experience) feel a little weird treating one's (a limited partner's) investment and the results of the sale as one's "purchase price" and "sales price," but I believe that's the correct way to think about it.
