Learning better balance after walking