Help?

I'm going back to work soon my baby is two months and I'm honestly so scared I've been EBF I've pumped a few times but I usually only get 3oz . So my question is what can I do to fit in pumping and what I can do to boost up my supply because I'm worried that when I go back I'll think I left enough milk and then find out it wasn't enough .. Sorry FTM !