Added comments and refactored an unnecessary method call.
Removed unused class DataMiningAssignment that was created upon project creation.