I actually started using Medela nipple cream. It helps a lot!
Yes cocoa butter is the best! Not cocoa butter lotion but 100% cocoa butter, I get the one that comes in a yellow stick. It's amazing lol.
Some cocoa butter can really help. Especially if they're drying up.