Doc string update and better readme