Added setNeedsLayout call