Diagonal augmentation added. Issue #8 fixed.