Make it work on small screens