small change
not fully complete check ... could'nt work it out properly