One final mod to solution 4