refaktoring + add komment