Added changes because of balancing