added balance check feature