update readme and add predict method