Betta pellets should help. I don't have shrimp, so I can not relate to the issues you have exactly, but if most of the food is sinking to the substrate before your Betta gets it (which I imagine the flakes are doing?), I would try a food that floats more before sinking, which the Betta pellets...