Updated code to match window resize better

Still need to account to account for width downsizing not working
properly