Any advice on getting baby into crib?

My son is 2 months old and has allergy and reflux issues the first month and a half. So has has slept with us and takes naps in the swing. He's feeling better and we want to start transitioning into the crib. He doesn't hate it, but also doesn't fall asleep or stay asleep in there. Any tips? Should I be waiting until 3 months ?