UI fix for small screen sizes