Cleaned up the preview screen