Code P is for excess contributions to a 401K, so use code P. The IRS also receives a copy of your 1099-R and will be matching the information on your 1099-R to the information on your 2016 tax return, so input exactly what is on your 1099-R.
