Update/refactor Code component