updated code for better mobile UI/UX