made some small changes, moved KSPSetOperator to Mode::Setup