Looking really good! IMO I think a puffy mouth is really just a sign of it being “hungry” and is very different than a gaping mouth which is something to be worried about. Before you move it to the display you may want to try to feed it something meaty, like a small (dime-size) piece of fresh de-shelled uncooked shrimp to test things out before the Big Move! I always make sure my shrimp is fresh since a spoiled piece could put you right back at square-one in regards to treatment. I’ve tried pellets but my nems never go for them and just spit them out.