I'm going crazy!

My son is one week shy of 5 months. He's starting to master rolling over and has recently switched over to a crib because of this. He was swaddled and we transitioned out as soon as he became Houdini and rolling . So the issue is that he doesn't sleep! He used to be a decent sleeper. From birth he would Wake about two times a night for food and as he got older it went to one time and even had moments where he would sleep through the night. Ever since he started learning to roll his sleeping habits have drastically changed and got even worse now that he can roll on his stomach. He doesn't know how to roll back however. He also loves sleeping on his side, so when we do put him down for bed he rolls onto his side. So now when he falls asleep somewhere between 9:30 and 11:00 , he's up again by one and then 4 and then 6 or 7. Last night was the worst yet and he fell asleep by 10. I laid down for the night and he was up by 11:30 and awake till 12:40 and then woke again by 1 and up until 3....and then stirred most of the night until 7. The last couple of days he's woken himself up by turning on his stomach and before that he woke because of moving from side to side. I'm going crazy because I'm not sure what to do...and I also questioned on if it was because he was switched to the crib but he had two good nights in the crib where he slept through the night in the beginning. We started cereal with his milk and that's not helping. We even tried swaddling him again at first to see if that was it but nope and now we definitely won't since he gets on his stomach. I'm so exhausted and lost. Anyone have any tips or suggestions... I'm also a ftm so I'm new to everything lol....thanks!