Fixed wrong logic in rescaling events to account for the removed training set

It addresses issue #3. However, the commit does not fix it completely.
Finally fixes #3

I did not test the code.