UPDATES TO CODE