Added stuff to make it work better on mobile