game goes to charter after ending song from charter

source/funkin/editors/charter/Charter.hx

@@ -607,7 +607,6 @@ class Charter extends UIState {
 						: CoolUtil.floorInt((mousePos.y - dragStartPos.y) / 40);
 					var horizontalChange:Int = CoolUtil.floorInt((mousePos.x - dragStartPos.x) / 40);
 					var changePoint:FlxPoint = FlxPoint.get(verticalChange, horizontalChange);
-					trace(dragStartPos, hoverOffset,changePoint);
 
 					for (s in selection) {
 						if (!s.draggable) continue;

source/funkin/editors/charter/CharterSelection.hx

@@ -129,6 +129,7 @@ class CharterSelection extends EditorTreeMenu {
 
 		// Add to List
 		freeplayList.songs.insert(0, creation.meta);
+		@:privateAccess main.preAdd(option);
 		main.members.insert(1, option);
 		// @:privateAccess main.__firstFrame = true;
 	}

source/funkin/game/PlayState.hx

@@ -1338,8 +1338,10 @@ class PlayState extends MusicBeatState
 		}
 		else
 		{
-			trace('WENT BACK TO FREEPLAY??');
-			FlxG.switchState(new FreeplayState());
+			if (chartingMode)
+				FlxG.switchState(new funkin.editors.charter.Charter(SONG.meta.name, difficulty, false));
+			else
+				FlxG.switchState(new FreeplayState());
 		}
 	}