adding imgs + improvemenement