Add basic size inference.

Noted some issues....
Merge pull request #1 from a-wortman/size_checking

Add basic size inference.