Just two months after giving birth to her daughter, Stormi Webster, Kylie Jenner is already focusing on her post-baby body. The 20-year-old has been sharing throwback photos as inspiration and has already begun working out.

“She’s focusing on getting her body back and also getting back to a comfortable weight aside from caring for Stormi,” a source told People, adding that the reality star “wasn’t the most comfortable while pregnant.”

“Kylie wants to be a really hot young mom and is making sure that happens,” the source said, adding that as she gets further along in her transformation, she will be “flaunting her progress more and more.”

Last week, the Lip Kit mogul took to Instagram to reveal her secret for weight loss. “My girl @premadonna87 hooked me up with the @waistgangsociety snap back package,” she wrote in the caption of a photo of her wearing a waist trainer. “Waistgang has the BEST quality snapback products.”

But she got a ton of backlash from fans online for promoting waist trainers because many deem it as an unhealthy way to lose weight. “My god is Kylie still advertising waist trainers (NOW POST GIVING BIRTH) to teenage girls,” one fan wrote. “How can u be so filthy rich and still sell out for something that is so dumb and could potentially perpetuate negative body image in young girls.”

Insiders revealed that Kylie gained about 50 pounds during her pregnancy and at her heaviest, she weighed 180 pounds. She enjoyed herself while she was “eating for two” and often indulged in her pregnancy cravings — which included lots of In-N-Out. But in just one month after giving birth to her daughter, Kylie lost an impressive 35 pounds but it seemed like she put a lot of pressure on herself to do so in such a short amount of time.

“She called herself ‘fat,’ ‘gross,’ and ‘disgusting’ all the time,” an insider told Life & Style exclusively. “There was no way Kylie wanted to be seen in public. Now she feels like she’s got a lot to prove.”