Readme update + minor bug fixed