polished code thanks for previous one

THATS THE POLISH ONE