fixed register model and updated readme