Refactored, cleaned, and faster. Better README. Included datasets.