Add method to adjust height for small screen