updated button for smallscreen