Rob Kardashian is ready to get the baby weight off