Dia is back at work after embracing motherhood