* Add SizedBox.square() constructor * Fix comment and move test case to the right file * Add missing import statement