improving image according to Kim comments
small iteration in the first part