Remove count dict
Fix directory error