Removed unused parts of script
Script uses multinomialnb
Improved memory usage of script