Finish validation, add balance check