Removed unused imports
Change default input file name
Support getting number of cluster from user
Memory optimization