Fixed the imaging problems