A fun piece, by all means @Samuel_vangogh!
I'm not exactly an expert on chamber music (let alone the viola), but I personally find the viola occupies an almost uncanny middle-ground between the violin and the cello in this piece. I suppose this relays my biases towards the violin and cello, but I think it'd definitely benefit if you stuck with playing around on either the higher or the lower register that either instrument provides (I think it kind of sounds like you're playing around middle C on the piano sometimes).
On the other hand (aside from the incomplete piano in the first theme), I think the themes blend with each other reasonably well: it makes for an interesting slow-fast-slow-type movement throughout the piece, like what Schubert did in some of his lieder. I'll say, though, that developing the ideas more would really take it that extra mile (there's obviously that abrupt end, which I believe was intentional).
I am aware this is still a sketch, but those are my thoughts on how you might want to complete the piece. Keep at it! 💪