Yeah, you sound like you're right about to get it down. Just take every spare moment to practice. When I was learning I would make myself try to trackstand as long as I could whenever I reached my destination. That way if you bail you just lock up your bike, you meant to get off.