Improve small screen styles