Fixed the editor to allow it to be nicer