fixed typo
more efficient distance computation
Removed debug plot from code
Added load function, documentation and cleaned up
Extra fields in data class for convenience