Removing puts
Overriding checkout flow