Add basic size inference.

Noted some issues....