code reviewed

chanded code made it lighter