Trying to extend Checkout controller