improve balance

improve balance