Slight modifications to the rainfall model