layout fix for mobile and desktop