Code cleanup
Implemented Model#total_count