UI changes and trying to fix the mobile css bug