Update collapse.js

set `width` to `innerWidth` instead of `innerHeight`