Updates

o readme new figures
o small changes to class