Made it work for smaller screens