Code cleanup, added proper image.