Rework cancel() method
Forgot header file