From what I’ve read sleep can be one of the things that takes the longest to return to normal. Honestly just going to bed around 10 and getting up around 6 along with diet, and lifting while just giving it time is the best thing you can do. One thing that’s helped with insomnia and getting to fall asleep is CBD gummies that I take right before bed and it knocks me out most of the time. That and meditating right before bed helps, but the REM thing is sort of out of your control. Try and control what you can control and put your body in the best position to heal itself (Albizia didn’t really do anything for me either btw)