Update in checkout controller (still not fully implemented)