Fixing layout issue on mobile