Improve block preconditioner in LLG class