balancing
some code changes