Add finish state & the logic to calculate the stars