Fix plotting issue
Update readme