Add support for window resize