Added screen size fix