Improve Block preview on page edit screen