added images / fixed code