Added check screen size javascript back in