Cleaned up code. returned back to original images.