Removed unused import
Implement data directory path
Ignore PyCharm IDE files