Kanye’s New Year: West ‘Doing Much Better’ After Flying with Kim and the Kids to Visit Mom’s Grave

Kanye West continues to recover following his recent hospitalization for exhaustion by spending time with family.

The 39-year-old rapper flew to Oklahoma City on Sunday with wife Kim Kardashian West, 36, and their two children —  North, 3½, and Saint, 1 — to visit his mother’s grave, according to paparazzi. A source tells PEOPLE that after spending New Year’s Eve together, the family departed on a private jet from Van Nuys airport on Sunday morning and returned to their Bel Air house together Monday evening.

“Kim and Kanye spent New Year’s Eve together with the kids. Kanye is doing much better. He has worked out every day and is taking care of himself. Kanye seems much better mentally. He is sleeping better and seems much healthier,” says the insider.

The source continued, “He and Kim are getting along better. Things are looking better for the new year.”

A different source previously told PEOPLE that the couple’s marriage has been under strain since West’s health crisis hit — less than two months after his wife’s terrifyingrobbery at gunpoint in Paris. However, the family attended Kris Jenner‘s star-studded Christmas Eve party, even posting a photo of the four in front of an enormous Christmas tree from the bash.

“They were focused on making it a fun Christmas for the kids,” said the source. “They hang out with the kids, but are not really getting along otherwise.”

Despite the marital frustrations, the couple have continued to step out together here and there, including for date night in Santa Monica on Dec. 18 and most recently to catch the Los Angeles Ballet’s The Nutcracker with daughter North.

West has been spotting several times heading to the movies. The rapper was spotted leaving a movie theater the day after Christmas rocking army pants, a shearling jacket and a new dash of pink in his recently dyed blond hair.