more functionality and better readme