Return correct value in SAESOptimizer
Change CSS for documentation. Improve docstrings
Implement image plot. Add some docstrings