Clean up. Removed old functions
Wrote print_counts function