Changed size method to count on total_records