Can you remove the localStorage logic now?