Add call to method to end of code