added balance tests