#4 fix and fix readme