I forgot. Small changes. Making one to work with a_star.