Now works better on small screens