Kylie Jenner is "focusing" on getting her pre-pregnancy body back as she wants to be at a more "comfortable weight".