Add payment capture functionality