Added in functionality for drift calculation