Refactor rheology calculator - make it really virtual, override properly, etc