FINISHED the not-efficient answer, but failed to finished the follow-up Problem