Added module initialization
Removed class files
Updated main.py to include new directory paths