Yes you can, just make sure you take the relevant documentation related to company B to present at the port of entry such as original I-797, offer letter etc.