bug fixes
Fixed sizing