Button change to work on small screens