added balance but need to fix