small ui and balance improvements