On the input screen for 1099-R under Income > Pensions, IRAs (1099-R), there are two lines for distribution codes.  Within those dropdown boxes, you'd find the codes for both.

Here's also an article on how to make your entries for 1099-R: https://proconnect.intuit.com/support/en-us/help-article/retirement-benefits/common-questions-form-1...