add straighten function