Fixing styling issues for mobile