Fixes K-Means trainer display issue and closes #133