added possibility to edit existing code