Minor changes to under-relaxation