Done. Need to add comments and compare