Update readme. Impove tests