working on balance handling