Updating Readme and fixing an import bug