fix coloring
add rmse figs
improve reprs
refactor data structures into separate file
flake8 compliance