code per Yo + refactor