Editing and adding labels now possible, but still a bit buggy tho