Changed a bit of layout so looks better on mobile