Upravená logika - přidán __construct