added balance state test