Add resize functionality
Small fix