Bugfix + Updated Edit Mode