Add code that was removed