Added balance check