* Add SizedBox.square() constructor

* Fix comment and move test case to the right file

* Add missing import statement