Clean up

Signed-off-by: simonewebdesign <hello@simonewebdesign.it>
Fix edit bug