Definitely keep doing it as long as your baby likes it, but all babies are different. My friends bub would go to sleep almost instantly when he was wrapped, and was happy with it until he was about 8 months.
My little man hated it. Literally. After he was about 2 weeks old, it didn't matter how loose or tight I wrapped him, where his arms were or what position, he would fight and scream until he had his arms out and free. If he couldn't get them out, he would become almost hysterical (not that I left him too long to try, but you know). So I gave up on it.
He's almost 2 now, and still the same: 90% of the time, he will not go to sleep if he has a blanket on him. I can put it on him after he's asleep, but not before, he'll fight it and kick it off. Why, I don't know, he'll keep it on when he stirs during the night. Maybe he's just weird.
Anyway, keep doing what works for you and your bub!! And good luck!!