added a better ending