work on checkout screen