Solution: add automatic charge balancing and other misc improvements

Solution: add automatic charge balancing and other misc improvements #460