Button fix for small screens