Add window resize code