Just a simple modification